import requests

from common.get_header_tools import get_header


class IhrmEmpApi(object):
    # 添加员工
    def add_emp(self,req_header,req_body):
        resp = requests.post(url="http://ihrm2-test.itheima.net/api/sys/user",
                             headers=req_header,
                             json=req_body)
        return  resp
    # 查询员工
    def query_emp(self,emp_id,req_header):
        resp = requests.get(url="http://ihrm2-test.itheima.net/api/sys/user/" + emp_id,
                             headers=req_header)
        return resp
    # 修改员工
    def update_emp(self,emp_id,req_header,req_body2):
        resp = requests.put(url="http://ihrm2-test.itheima.net/api/sys/user/" + emp_id,
                             headers=req_header,
                             json=req_body2)
        return resp
    # 删除员工
    def delete_emp(self,emp_id,req_header):
        resp = requests.delete(url="http://ihrm2-test.itheima.net/api/sys/user/" + emp_id,
                             headers=req_header)
        return resp

if __name__ == '__main__':
    ihrm = IhrmEmpApi()
    # 准备请求头 - 调用我们自己封装的 工具方法
    req_header = get_header()
    req_body = {
        "username":"刘宏飞298",
        "mobile":"18611222222",
        "workNumber":"11222222"
    }
    # 调用添加员工方法
    resp = ihrm.add_emp(req_header,req_body)
    # 打印添加员工结果
    print("添加员工：",resp.json())
    # 调用查询员工方法
    emp_id = "1609853550223962112"
    resp2 = ihrm.query_emp(emp_id,req_header)
    print("查询结果：",resp2.json())
    # 调用修改员工方法
    req_body2 = {
        "username": "刘宏飞188"
    }
    resp3 = ihrm.update_emp(emp_id,req_header,req_body2)
    print("修改员工：", resp3.json())
    # 调用删除员工方法
    resp4 = ihrm.delete_emp(emp_id,req_header)
    print("删除员工：",resp4.json())
